package com.tanhua.mongo.api;

import com.tanhua.api.service.UserLikeApi;
import com.tanhua.model.mongo.UserLike;
import org.apache.dubbo.config.annotation.DubboService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.data.mongodb.core.query.Update;

import java.util.List;
import java.util.stream.Collectors;

@DubboService
public class UserLikeApiImpl implements UserLikeApi {

    @Autowired
    private MongoTemplate mongoTemplate;

    @Override
    public List<Long> findByUserId(Long userId) {
        return mongoTemplate.find(Query.query(
                Criteria.where("userId").is(userId)
        ), UserLike.class)
                .stream().map(UserLike::getLikeUserId)
                .collect(Collectors.toList());
    }

    @Override
    public Boolean saveUserLikeByLove(Long userId, Long likeUserId, boolean isLike) {

        try {

            Query query = Query.query(Criteria
                    .where("userId").is(userId)
                    .and("likeUserId").is(likeUserId));
            // 查询数据是否存在
            UserLike userLike = mongoTemplate.findOne(query, UserLike.class);

            if (userLike == null) {
                // 没有数据，添加数据
                userLike = new UserLike();
                userLike.setUserId(userId);
                userLike.setLikeUserId(likeUserId);
                userLike.setIsLike(isLike);
                userLike.setCreated(System.currentTimeMillis());
                userLike.setUpdated(System.currentTimeMillis());

                mongoTemplate.save(userLike);
            } else {
                // 表示不喜欢，修改数据为喜欢
                Update update = Update.update("isLike", isLike)
                                .set("updated", System.currentTimeMillis());
                mongoTemplate.updateFirst(query, update, UserLike.class);
            }

            return true;
        } catch (Exception e) {
            e.printStackTrace();
        }

        return false;
    }
}
